<template>
  <div class="app-container">
    <el-form
      :model="queryParams"
      ref="queryForm"
      :inline="true"
      v-show="showSearch"
      label-width="100px"
    >
      <el-form-item label="企业类型" prop="enterpriseType">
        <el-input
          v-model="queryParams.enterpriseType"
          placeholder="请输入企业类型"
          clearable
          size="small"
          @keyup.enter.native="handleQuery"
        />
      </el-form-item>
      <el-form-item>
        <el-button
          type="primary"
          icon="el-icon-search"
          size="mini"
          @click="handleQuery"
          >搜索</el-button
        >
        <el-button icon="el-icon-refresh" size="mini" @click="resetQuery"
          >重置</el-button
        >
      </el-form-item>
    </el-form>

    <el-row :gutter="10" class="mb8">
      <el-col :span="1.5">
        <el-button
          type="primary"
          plain
          icon="el-icon-plus"
          size="mini"
          @click="handleAdd"
          v-hasPermi="['energy:qyEnterpriseType:add']"
          >新增</el-button
        >
      </el-col>
      <el-col :span="1.5">
        <el-button
          type="success"
          plain
          icon="el-icon-edit"
          size="mini"
          :disabled="single"
          @click="handleUpdate"
          v-hasPermi="['energy:qyEnterpriseType:edit']"
          >修改</el-button
        >
      </el-col>
      <el-col :span="1.5">
        <el-button
          type="danger"
          plain
          icon="el-icon-delete"
          size="mini"
          :disabled="multiple"
          @click="handleDelete"
          v-hasPermi="['energy:qyEnterpriseType:remove']"
          >删除</el-button
        >
      </el-col>
      <el-col :span="1.5">
        <el-button
          type="warning"
          plain
          icon="el-icon-download"
          size="mini"
          @click="handleExport"
          v-hasPermi="['energy:qyEnterpriseType:export']"
          >导出</el-button
        >
      </el-col>
      <right-toolbar
        :showSearch.sync="showSearch"
        @queryTable="getList"
      ></right-toolbar>
    </el-row>

    <el-table
      v-loading="loading"
      :data="qyEnterpriseTypeList"
      @selection-change="handleSelectionChange"
      row-key="id"
      default-expand-all
      :tree-props="{ children: 'children', hasChildren: 'hasChildren' }"
    >
      <el-table-column type="selection" width="55" align="center" />
      <el-table-column label="企业类型" prop="enterpriseType" />
<!--      <el-table-column label="污水处理费" align="center" prop="feeAmount" />-->
      <el-table-column label="备注" align="center" prop="remark" />
      <el-table-column
        label="操作"
        align="center"
        class-name="small-padding fixed-width"
      >
        <template slot-scope="scope">
          <el-button
            size="mini"
            type="text"
            icon="el-icon-edit"
            @click="handleUpdate(scope.row)"
            v-hasPermi="['energy:qyEnterpriseType:edit']"
            >修改</el-button
          >
          <el-button
            size="mini"
            type="text"
            icon="el-icon-delete"
            @click="handleDelete(scope.row)"
            v-hasPermi="['energy:qyEnterpriseType:remove']"
            >删除</el-button
          >
        </template>
      </el-table-column>
    </el-table>

    <!-- <pagination
      v-show="total > 0"
      :total="total"
      :page.sync="queryParams.pageNum"
      :limit.sync="queryParams.pageSize"
      @pagination="getList"
    /> -->

    <!-- 添加或修改企业类型对话框 -->
    <el-dialog :title="title" :visible.sync="open" width="540px" append-to-body>
      <el-form ref="form" :model="form" :rules="rules" label-width="90px">
        <el-form-item label="企业类型" prop="enterpriseType">
          <el-input
            v-model="form.enterpriseType"
            placeholder="请输入企业类型"
          />
        </el-form-item>

        <el-form-item label="父级" prop="parentId">
          <treeselect v-model="form.parentId" :options="qyEnterpriseTypeOptionList" :normalizer="normalizer"
                      placeholder="请选择父级"/>
        </el-form-item>

<!--        <el-form-item label="污水处理费" prop="feeAmount">-->
<!--          <el-input v-model="form.feeAmount" placeholder="请输入污水处理费" />-->
<!--        </el-form-item>-->
        <el-form-item label="备注" prop="remark">
          <el-input
            v-model="form.remark"
            type="textarea"
            placeholder="请输入内容"
          />
        </el-form-item>
      </el-form>
      <div slot="footer" class="dialog-footer">
        <el-button type="primary" @click="submitForm">确 定</el-button>
        <el-button @click="cancel">取 消</el-button>
      </div>
    </el-dialog>
  </div>
</template>

<script>
import {
  listQyEnterpriseType,
  getQyEnterpriseType,
  delQyEnterpriseType,
  addQyEnterpriseType,
  updateQyEnterpriseType,
  getQyEnterpriseTypeTree,
} from "@/api/smartParks/merchants/enterpriseType";
import Treeselect from "@riophae/vue-treeselect";
import "@riophae/vue-treeselect/dist/vue-treeselect.css";
export default {
  name: "QyEnterpriseType",
  components: {
    Treeselect
  },
  data() {
    return {
      deptOptions: [],
      defaultProps: {
        children: "children",
        label: "enterpriseType",
      },
      // 遮罩层
      loading: true,
      // 选中数组
      ids: [],
      // 非单个禁用
      single: true,
      // 非多个禁用
      multiple: true,
      // 显示搜索条件
      showSearch: true,
      // 总条数
      total: 0,
      // 企业类型表格数据
      qyEnterpriseTypeList: [],
      qyEnterpriseTypeOptionList:[],
      // 弹出层标题
      title: "",
      // 是否显示弹出层
      open: false,
      // 查询参数
      queryParams: {
        enterpriseType: null,

      },
      // 表单参数
      form: {
        parentId:null
      },
      // 表单校验
      rules: {
        enterpriseType: [
          {
            required: true,
            message: "请输入企业类型",
            trigger: "blur",
          },
        ],
      },
    };
  },
  created() {
    this.getList();
    this.getTree('');
  },
  methods: {
    getTree(id) {
      getQyEnterpriseTypeTree(id).then((res) => {
        this.deptOptions = res.data ? res.data : [];
      });
    },
    handleNodeClick(data) {
      this.form.parentId = data.id;
    },
    /** 查询企业类型列表 */
    getList() {
      this.loading = true;
      listQyEnterpriseType(this.queryParams).then((response) => {
        this.qyEnterpriseTypeList = this.handleTree(response.rows, "id");
        this.qyEnterpriseTypeOptionList=JSON.parse(JSON.stringify(this.qyEnterpriseTypeList))
        this.qyEnterpriseTypeOptionList.forEach(item=>{
          item.children=[]
        })
        // 把树形只保留两级
        this.loading = false;
      });
    },
    normalizer(node) {
      if (node.children && !node.children.length) {
        delete node.children;
      }
      return {
        id: node.id,
        label: node.enterpriseType,
        children: node.children
      };
    },
    optionTree(treeList,lv){
      if (this.isEmpty(treeList)){
        return
      }
      treeList.forEach((item,index)=>{
        if (lv==2){
          item.children=[]
        }else if (!this.isEmpty(item.children)){
          this.optionTree(item.children,lv+1)
        }
      })
    },
    // 取消按钮
    cancel() {
      this.open = false;
      this.reset();
    },
    // 表单重置
    reset() {
      this.form = {
        id: null,
        enterpriseType: null,
        feeAmount: null,
      };
      this.resetForm("form");
    },
    /** 搜索按钮操作 */
    handleQuery() {
      this.getList();
    },
    /** 重置按钮操作 */
    resetQuery() {
      this.resetForm("queryForm");
      this.handleQuery();
    },
    // 多选框选中数据
    handleSelectionChange(selection) {
      this.ids = selection.map((item) => item.id);
      this.single = selection.length !== 1;
      this.multiple = !selection.length;
    },
    /** 新增按钮操作 */
    handleAdd() {
      this.reset();
      this.open = true;
      this.title = "添加企业类型";
    },
    /** 修改按钮操作 */
    handleUpdate(row) {
      this.reset();
      const id = row.id || this.ids;
      getQyEnterpriseType(id).then((response) => {
        this.form = response.data;

        this.getTree(this.form.id);
        this.open = true;
        this.title = "修改企业类型";
      });
    },
    /** 提交按钮 */
    submitForm() {
      this.$refs["form"].validate((valid) => {
        if (valid) {
          if (this.form.id != null) {
            updateQyEnterpriseType(this.form).then((response) => {
              this.msgSuccess("修改成功");
              this.open = false;
              this.getList();
            });
          } else {
            addQyEnterpriseType(this.form).then((response) => {
              this.msgSuccess("新增成功");
              this.open = false;
              this.getList();
            });
          }
        }
      });
    },
    /** 删除按钮操作 */
    handleDelete(row) {
      const ids = row.id || this.ids;
      this.$confirm("是否确认删除企业类型编号的数据项?", "警告", {
        confirmButtonText: "确定",
        cancelButtonText: "取消",
        type: "warning",
      })
        .then(function () {
          return delQyEnterpriseType(ids);
        })
        .then(() => {
          this.getList();
          this.msgSuccess("删除成功");
        });
    },
    /** 导出按钮操作 */
    handleExport() {
      this.download(
        "energy/qyEnterpriseType/export",
        {
          ...this.queryParams,
        },
        `energy_qyEnterpriseType.xlsx`
      );
    },
  },
};
</script>
